package org.algorithm.二师兄.数组.三月.第二周;

import java.util.Arrays;

/**
 * @author 二师兄
 * @date 2022/03/08 11:44 PM
 * @description:       周五作业
 *
 *  <p>
 *      输入：nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
 *      输出：[1,2,2,3,5,6]
 *      解释：需要合并 [1,2,3] 和 [2,5,6] 。
 *      合并结果是 [1,2,2,3,5,6] ，其中斜体加粗标注的为 nums1 中的元素。
 *  </p>
 *
 * uri:  https://leetcode-cn.com/problems/merge-sorted-array/
 */
public class 合并两个有序数组 {

    public static void main(String[] args) {
        int[] nums = {1,2,3,0,0,0};
        int[] arr = {2,5,6};
        merge1(nums, nums.length, arr, arr.length);
        System.out.println(Arrays.toString(nums));
    }


    public static void merge(int[] nums1, int m, int[] nums2, int n) {
        // 第一种办法     短数组直接加入到后面,   再排序
        if (m == 0 || n == 0) return;
//        for (int i = 0; i < n; i++) {
//            nums1[m + i] = nums2[i];
//        }
//        Arrays.sort(nums1);

        // 第二种
// i、k 分别指向 nums1 和 nums2 中初始化的元素的最后一位
        // j 指向 nums1 的末尾处
        int i = m - 1, j = m + n - 1, k = n - 1;

        while (i >= 0 && k >= 0) {
            // 从后往前更新 nums1
            if (nums2[k] >= nums1[i]) {
                // 将 nums1[j] 的值更新为 nums2[k]
                nums1[j] = nums2[k];
                --j;
                --k;
            } else {
                // 将 nums1[j] 的值更新为 nums1[i]
                nums1[j] = nums1[i];
                --j;
                --i;
            }
        }

        // 将 nums2 从 0 到 k 的部分添加进 nums1 的头部
        System.arraycopy(nums2, 0, nums1, 0, k + 1);
    }

    public static void merge1(int[] nums1, int m, int[] nums2, int n) {
        // i、k 分别指向 nums1 和 nums2 中初始化的元素的最后一位
        // j 指向 nums1 的末尾处
        int i = m - 1, j = m + n - 1, k = n - 1;

        while (i >= 0 && k >= 0) {
            // 从后往前更新 nums1
            if (nums2[k] >= nums1[i]) {
                // 将 nums1[j] 的值更新为 nums2[k]
                nums1[j] = nums2[k];
                --j;
                --k;
            } else {
                // 将 nums1[j] 的值更新为 nums1[i]
                nums1[j] = nums1[i];
                --j;
                --i;
            }
        }

        // 将 nums2 从 0 到 k 的部分添加进 nums1 的头部
        System.arraycopy(nums2, 0, nums1, 0, k + 1);
    }
}
